Learning from What You Retrieve: Online RL Fine-Tuning for Semantic Retrieval
从检索结果中学习:语义检索的在线 RL 微调
🧭 语义检索 · 在线 RL 微调 ★ 深度解读
Alibaba Group
提出 PAO,在冻结文档索引下只对正 advantage 样本做 RL 更新,避免 embedding 几何塌缩并提升检索效果
📌 工作参考价值(4 点)
- 工业双塔召回中文档索引冻结是常态,PAO 的 RL 更新只作用于 query encoder,无需重建索引;可直接改造现有 REINFORCE:对 advantage 加 I(A>0) mask,并加入 KL 到预训练 reference policy,工程落地成本低。
- 如果业务上用 reranker 或点击信号作为 reward 做在线 RL,不要无差别惩罚低 reward 样本。标准 policy gradient 的负向 push 会破坏预训练语义流形;优先只利用正 reward 的 pull 信号,能更稳定地做 retriever-reranker 对齐。
- 与直接蒸馏 reranker soft distribution 相比,positive-only policy gradient 在冻结索引下更稳,尤其深层召回 Recall@50 提升更明显;当目标是对齐精排偏好时,可优先考虑 PAO 而非全局 KL 蒸馏。
- 使用标准化 advantage 降低方差;β=0.1~0.3、τ≈1.0 是较稳健的超参区间。注意 reward model 的偏差会传导到召回,若 reranker 偏好标题党,需要多目标或额外约束。